算子库支持情况

  • MindSpore Lite支持不同硬件后端的算子列表

算子名称

CPU

FT04

FT78

PNNA

Abs

FP16
FP32
Int32
Int8
UInt8

FP32
Int8

AbsGrad

FP32

Activation

FP16
FP32
Int32
Int8
UInt8

FP32
Int8

ActivationGrad

FP16
FP32

Adam

FP32

AddFusion

FP16
FP32
Int32
Int8
UInt8
Bool

FP16
FP32
Int32
Int16
Cplx64

FP32
FP64
Int8
Int32
Int16
Cplx64
Cplx128

FP32
Int8

AdderFusion

FP32

AddGrad

FP32

AddN

FP16
FP32

FP32
Int8

Affine

FP32

All

FP32

AllGather

FP32

ApplyMomentum

FP32

FP16
FP32

FP32

Assert

FP16
FP32
Bool

Int32

Int32

Assign

FP32

ArgmaxFusion

FP16
FP32
Int8
UInt8

FP32
Int8

ArgminFusion

FP16
FP32
Int8
UInt8

FP32
Int8

AvgPoolFusion

FP16
FP32
Int8
UInt8

FP32
Int8

Attention

-

FP32

-

AvgPoolGrad

FP16
FP32

BatchNorm

FP16
FP32
Int8
UInt8

BatchNormGrad

FP16
FP32

BatchToSpace

FP32
Int8
UInt8

FP32
Int8

BatchToSpaceND

FP16
FP32
Int8
UInt8

BiasAdd

FP16
FP32
Int8
UInt8

BiasAddGrad

FP16
FP32

BinaryCrossEntropy

FP32

BinaryCrossEntropyGrad

FP32

BroadcastTo

FP16
FP32
Int32
Bool

Call

FP16
FP32
Int32
Bool

Cast

FP16
FP32
Int32
Int8
UInt8
Bool

Ceil

FP16
FP32
Int8
UInt8

FP32
Int8

Clip

FP32
Int32

FP32
Int8

Concat

FP16
FP32
Int32
Int8
UInt8
Bool

FP32
Int8

ConstantOfShape

FP16
FP32
Int32

Conv2DFusion

FP16
FP32
Int8
UInt8

FP32
Int8

Conv2DBackpropFilterFusion

FP16
FP32

Conv2DBackpropInputFusion

FP16
FP32

Conv2dTransposeFusion

FP16
FP32
Int8
UInt8

FP32
Int8

Cos

FP16
FP32
Int8
UInt8

Crop

FP16
FP32
Int32
Int8
UInt8

CropAndResize

FP32

CumSum

FP32
Int32

CustomExtractFeatures

FP32

CustomNormalize

FP32

CustomPredict

FP32
Int32

DEConv2DGradFilter

FP32

DepthToSpace

FP16
FP32
Int8
UInt8

FP32
Int8

DetectionPostProcess

FP32
Int8
UInt8

DivFusion

FP16
FP32
Int32
Int8
UInt8

FP16
FP32
Int32
Int16
Cplx64

FP32
FP64
Int8
Int32
Int16
Cplx64
Cplx128

FP32
Int8

DivGrad

FP32

Dropout

FP16
FP32

DropoutGrad

FP16
FP32

DynamicQuant

FP32

Eltwise

FP16
FP32
Int8
UInt8

Elu

FP16
FP32

Equal

FP16
FP32
Int32
Int8
UInt8

EmbeddingLookupFusion

FP32

Erf

FP16
FP32

ExpFusion

FP16
FP32

FP16
FP32
Int32
Int16
Cplx64

FP32
FP64
Int8
Int32
Cplx64
Cplx128

ExpandDims

FP16
FP32
Int32
Int8
UInt8
Bool

Fill

FP16
FP32
Int32
Bool

Flatten

FP16
FP32
Int32

FlattenGrad

FP16
FP32

Floor

FP16
FP32
Int8
UInt8

FloorDiv

FP16
FP32
Int32

FloorMod

FP16
FP32
Int32

FullConnection

FP16
FP32
Int8
UInt8

FusedBatchNorm

FP16
FP32
Int8
UInt8

GatherNd

FP16
FP32
Int32
Int8
UInt8
Bool

Gather

FP16
FP32
Int32
Int8
UInt8
Bool

GatherD

FP16
FP32
Int32
Bool

GLU

FP32

Greater

FP16
FP32
Int32
Int8
UInt8

GreaterEqual

FP16
FP32
Int32
Int8
UInt8

GroupNormFusion

FP32

GRU

FP16
FP32

HashtableLookup

FP32
Int32

InstanceNorm

FP16
FP32

InvertPermutation

FP16
FP32
Int32

IsFinite

FP32

L2NormalizeFusion

FP32
Int8
UInt8

LayerNormFusion

FP16
FP32
Int8

LayerNormGrad

FP16
FP32

LeakyReLU

FP16
FP32
Int8
UInt8

Less

FP16
FP32
Int32
Int8
UInt8

LessEqual

FP16
FP32
Int32
Int8
UInt8

LinSpace

-

FP16
FP32

FP32

-

LRN

FP32

Log

FP16
FP32
Int8
UInt8

Log1p

FP32

LogGrad

FP16
FP32

LogicalAnd

FP16
FP32
Int32
Bool

LogicalNot

FP16
FP32
Int8
UInt8
Bool

LogicalOr

FP16
FP32
Bool

LogSoftmax

FP16
FP32

LshProjection

FP32

LSTM

FP16
FP32

LSTMGrad

FP32

LSTMGradData

FP32

LSTMGradWeight

FP32

MatMulFusion

FP16
FP32
Int8
UInt8

FP16
FP32
Int32
Int16
Cplx64

FP32
FP64
Int8
Int32
Int16
Cplx64
Cplx128

Maximum

FP16
FP32
Int32

MaximumGrad

FP16
FP32

MaxPoolFusion

FP16
FP32
Int8
UInt8

FP32
Int8

MaxPoolGrad

FP16
FP32

Merge

FP16
FP32

Minimum

FP16
FP32
Int32

MinimumGrad

FP16
FP32

Mod

FP32
Int32

MulFusion

FP16
FP32
Int32
Int8
UInt8

FP16
FP32
Int32
Int16
Cplx64

FP32
FP64
Int8
Int32
Int16
Cplx64
Cplx128

FP32
Int8

MulGrad

FP32

Neg

FP16
FP32
Int32

FP32
Int8

NegGrad

FP16
FP32

NLLLoss

FP32

NLLLossGrad

FP32

NotEqual

FP16
FP32
Int32
Int8
UInt8

NonMaxSuppression

FP32

NonZero

Bool

OneHot

FP16
FP32
Int32

FP32
Int8

OnesLike

FP16
FP32
Int32

PadFusion

FP16
FP32
Int8
UInt8

FP32
Int8

PartialFusion

FP16
FP32
Int32
Bool

PowFusion

FP16
FP32
Int8
UInt8

PowerGrad

FP32

PriorBox

FP32
Int8
UInt8

PReLUFusion

FP16
FP32

QuantDTypeCast

FP16
FP32
Int8
UInt8

FP32
Int8

RaggedRange

FP16
FP32
Int32

FP16
FP32
Int16
Int32

FP32
FP64
Int8
Int16
Int32

-

RandomNormal

FP16
FP32

RandomStandardNormal

FP16
FP32

Range

FP16
FP32
Int32

FP16
FP32
Int16
Int32

FP32
FP64
Int8
Int16
Int32

-

Rank

FP16
FP32

RealDiv

FP16
FP32

Reciprocal

FP16
FP32
Int8

ReduceFusion

FP16
FP32
Int32
Int8
UInt8
Bool

ReduceScatter

FP32

Reshape

FP16
FP32
Int32
Int8
UInt8
Bool

Resize

FP16
FP32
Int8
UInt8

ResizeGrad

FP16
FP32

ReverseV2

FP32
Int32

ReverseSequence

FP32

ROIPooling

FP32

Round

FP16
FP32
Int8
UInt8

Rsqrt

FP16
FP32
Int8
UInt8

RsqrtGrad

FP32

Select

FP32
Bool

Selu

-

ScaleFusion

FP16
FP32
Int8
UInt8

ScatterNd

FP16
FP32
Int32

ScatterNdUpdate

FP16
FP32
Int32

SGD

FP32

Shape

FP16
FP32
Int32
Int8
UInt8
Bool

SigmoidCrossEntropyWithLogits

FP32

SigmoidCrossEntropyWithLogitsGrad

FP32

Sin

FP16
FP32
Int8
UInt8

Size

FP16
FP32
Int32

SliceFusion

FP16
FP32
Int32
Int8
UInt8

SkipGram

FP32

SmoothL1Loss

FP32

SmoothL1LossGrad

FP32

Softmax

FP16
FP32
Int8
UInt8

FP32
Int8

SoftmaxGrad

FP32

Softplus

FP16
FP32

SpaceToBatch

FP16
FP32
Int8
UInt8

SpaceToBatchND

FP16
FP32
Int8
UInt8

SpaceToDepth

FP16
FP32

SparseToDense

FP16
FP32
Int32

SparseSoftmaxCrossEntropyWithLogits

FP32

Splice

FP16
FP32

Split

FP16
FP32
Int32
Int8
UInt8

FP32
Int32

SplitWithOverlap

FP16
FP32

Sqrt

FP16
FP32
Int8
UInt8

SqrtGrad

FP32

Square

FP16
FP32
Int8
UInt8

SquaredDifference

FP16
FP32

Squeeze

FP16
FP32
Int32
Int8
UInt8
Bool

StridedSlice

FP16
FP32
Int32
Int8
UInt8

StridedSliceGrad

FP16
FP32

Stack

FP16
FP32
Int32

SubFusion

FP16
FP32
Int32
Int8
UInt8

FP16
FP32
Int32
Int16
Cplx64

FP32
FP64
Int8
Int32
Int16
Cplx64
Cplx128

FP32
Int8

SubGrad

FP32

Switch

FP16
FP32
Int32
Bool

SwitchLayer

FP16
FP32
Int32
Bool

TensorListFromTensor

FP16
FP32
Int32

TensorListGetItem

FP16
FP32
Int32

TensorListReserve

FP16
FP32
Int32

TensorListSetItem

FP16
FP32
Int32

TensorListStack

FP16
FP32
Int32

TensorScatterAdd

FP32
Int32

TileFusion

FP16
FP32
Int32
Bool

TopKFusion

FP16
FP32
Int32
Int8
UInt8

Transpose

FP16
FP32
Int32
Int8
Bool

FP32
Int8

UniformReal

FP32
Int32

Unique

FP16
FP32
Int32

UnsortedSegmentSum

FP16
FP32
Int32

Unsqueeze

FP16
FP32
Int32
Int8
UInt8
Bool

FP32
Int8

Unstack

FP16
FP32
Int32

Where

FP16
FP32
Int32
Bool

ZerosLike

FP16
FP32
Int32

  • MindSpore Lite支持不同硬件后端的自定义算子列表:

算子名称

FT04

FT78

IFFT

Cplx64

Cplx64

FFT

Cplx64

Cplx64